/*
现在有两种线程，氧 oxygen 和氢 hydrogen，你的目标是组织这两种线程来产生水分子。

存在一个屏障（barrier）使得每个线程必须等候直到一个完整水分子能够被产生出来。

氢和氧线程会被分别给予 releaseHydrogen 和 releaseOxygen 方法来允许它们突破屏障。

这些线程应该三三成组突破屏障并能立即组合产生一个水分子。

你必须保证产生一个水分子所需线程的结合必须发生在下一个水分子产生之前。

换句话说:

如果一个氧线程到达屏障时没有氢线程到达，它必须等候直到两个氢线程到达。
如果一个氢线程到达屏障时没有其它线程到达，它必须等候直到一个氧线程和另一个氢线程到达。
书写满足这些限制条件的氢、氧线程同步代码。

 

示例 1:

输入: "HOH"
输出: "HHO"
解释: "HOH" 和 "OHH" 依然都是有效解。
示例 2:

输入: "OOHHHH"
输出: "HHOHHO"
解释: "HOHHHO", "OHHHHO", "HHOHOH", "HOHHOH", "OHHHOH", "HHOOHH", "HOHOHH" 和 "OHHOHH" 依然都是有效解。
 

提示：

输入字符串的总长将会是 3n, 1 ≤ n ≤ 50；
输入字符串中的 “H” 总数将会是 2n 。
输入字符串中的 “O” 总数将会是 n 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/building-h2o
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

#include "../stdc++.h"

// condition_variable
class H2O {
private:
    int countO{0}; // 氧气计数
    int countH{0}; // 氢气计数
    std::mutex m{};
    condition_variable cv;
public:
    H2O() {
        
    }

    void hydrogen(function<void()> releaseHydrogen) {
        std::unique_lock<std::mutex> ul(m);
        cv.wait(ul, [this](){return this->countH < 2;}); // 氢气最大是2
        // releaseHydrogen() outputs "H". Do not change or remove this line.
        releaseHydrogen();
        ++countH;
        if (countH + countO == 3) {
            countH = 0;
            countO = 0;
        }
        cv.notify_all();
    }

    void oxygen(function<void()> releaseOxygen) {
        std::unique_lock<std::mutex> ul(m);
        cv.wait(ul, [this](){return this->countO < 1;}); // 氧气最大是1
        // releaseOxygen() outputs "O". Do not change or remove this line.
        releaseOxygen();
        ++countO;
        if (countH + countO == 3) {
            countH = 0;
            countO = 0;
        }
        cv.notify_one();
    }
};

// semaphore
class Semphore {
private:
    int _n;
    std::mutex _m;
    std::condition_variable _cv;
public:
    Semphore(int n) : _n(n) {}
    void wait() {
        std::unique_lock<std::mutex> ul(_m);
        if (!_n) {
            _cv.wait(ul, [this](){return _n;});
        }
        --_n;
    }
    void signal() {
        std::unique_lock<std::mutex> ul(_m);
        ++_n;
        _cv.notify_one();
    }
};
class H2O {
private:
    Semphore s_hIn{2};
    Semphore s_oIn{1};
    Semphore s_hBarrier{0};
    Semphore s_oBarrier{0};
public:
    H2O() {
        
    }

    void hydrogen(function<void()> releaseHydrogen) {
        s_hIn.wait();
        s_oBarrier.signal();
        s_hBarrier.wait();
        // releaseHydrogen() outputs "H". Do not change or remove this line.
        releaseHydrogen();
        s_hIn.signal();
    }

    void oxygen(function<void()> releaseOxygen) {
        s_oIn.wait();
        s_oBarrier.wait();
        s_oBarrier.wait();
        s_hBarrier.signal();
        s_hBarrier.signal();
        // releaseOxygen() outputs "O". Do not change or remove this line.
        releaseOxygen();
        s_oIn.signal();
    }
};